'''
Author: cw
Date: 2025-08-21
Description: ...
'''
from pydantic import BaseModel, Field


class ProcessTask(BaseModel):
    """创建任务model"""
    id: str = Field(..., description="任务ID")
    prompt: str = Field(..., description="任务标题")
    num: int = Field(1, description="生成图片数量")
    size: str = Field("512x512", description="生成图片尺寸")
    provider: str = Field("mota", description="图片生成器")
    model: str = Field("Qwen/Qwen-Image", description="图片生成模型")
    extra: dict = Field(None, description="额外参数")
